from PIL import Image
import os
output_images_path = "./new"



list = []
list1 = []
# 这里读取文件目录和文件名
def file_name(file_dir):
    for root, dirs, files in os.walk(file_dir):
        # print('root_dir:', root)  # 当前目录路径
        # print('sub_dirs:', dirs)  # 当前路径下所有子目录
        # print('files:', files)  # 当前路径下所有非目录子文件
        for i in files:
            res = file_dir + '/' + i
            # print
            list.append(res)
            # print(res)


def file_name1(file_dir):
    for root, dirs, files in os.walk(file_dir):
        # print('root_dir:', root)  # 当前目录路径
        # print('sub_dirs:', dirs)  # 当前路径下所有子目录
        # print('files:', files)  # 当前路径下所有非目录子文件
        for i in files:
            res1 = file_dir + '/' + i
            list1.append(res1)
            # print(list)
            # print(res1)
            # yield res1
            # res = os.path.join(file_dir,i)


#加载底图
def pic(base_img_url,tmp_img_url):
    base_img = Image.open(base_img_url)#加载背景
    # 可以查看图片的size和mode，常见mode有RGB和RGBA，RGBA比RGB多了Alpha透明度
    #  print base_img.size, base_img.mode
    # box = (60, 44, 130, 120)  # 底图上需要P掉的区域
    box=(60,44,220,200)
    #加载需要P上去的图片
    tmp_img = Image.open(tmp_img_url)#要粘贴的图可选择大小或整张
    #这里可以选择一块区域或者整张图片
    # region = tmp_img.crop((10,10,100,110)) #选择一块区域
    region = tmp_img#使用整张
    #使用 paste(region, box) 方法将图片粘贴到另一种图片上去.
    #注意，region的大小必须和box的大小完全匹配。但是两张图片的mode可以不同，合并的时候回自动转化。如果#需要保留透明度，则使用RGMA mode
    #提前将图片进行缩放，以适应box区域大小
    # region = region.rotate(180) #对图片进行旋转
    region = region.resize((box[2] - box[0], box[3] - box[1]))
    base_img.paste(region, box)
    base_img.show() # 查看合成的图片
    import random
    res = random.randint(1,1000)
    base_img.save(res +'.png') #保存图片


file_name('./img')
file_name1('./two')
# 实行拼接
for i,j in zip(list,list1):
    base_img = Image.open(i)
    box = (60, 44, 220, 200)
    # for j in list1:
    tmp_img = Image.open(j)
    print(i,j)
    region = tmp_img
    region = region.resize((box[2] - box[0], box[3] - box[1]))
    base_img.paste(region, box)
    downloaddir = 'new'
    lujin = os.getcwd()
    # print(lujin )
    # if not os.path.exists(downloaddir):
    #     os.mkdir(downloaddir)

        # 换路径
        # path_url = r'D:\python\二维码\new'
        # base_img.save(i+'.png')  # 保存图片

    base_img.save(output_images_path+'/'+i+'.png') # 保存图片



print(list)
print(list1)
# pic(file_name,file_name1)
